from csv import *

class ArchivoCSV():
  """
  Clase para manejar la info del archivo CSV
  
  Esta clase contiene ...
  """
  def __init__(self, data, sep, tienehead):        
    self.tienehead = tienehead
    self.sep = sep.encode('ascii')
    #print self.sep
    # TODO: Mejorar esto
    datos = data.split('\n') 
    #Arregla el tema de que no agarre el head
    if self.tienehead:
        datos = datos[1:len(datos)]
    long = len(datos)    
    if (len(datos[long-1]) == 0):
      datos = datos[0:long-1]
    self.data = datos
    #print data
    self.t = []
    self.s = []
    cont = 0
    mir = reader( self.data, delimiter=self.sep, quoting=QUOTE_NONE )
    # 2 columnas
    for row in mir:
      # TODO: Provisorio
      # TODO: Separador decimal
      tt = row[0]
      ss = row[1] 
      if tt.count(',')>0:
        tt = tt.replace(',', '.')
      if ss.count(',')>0:
        ss = ss.replace(',', '.')        
      self.t.append( float(tt) )
      self.s.append( float(ss) )
      cont = cont + 1
    self.nr = cont

  def getData(self):
    return (self.data)
    
  def getTiempo(self):
      return ( self.t )

  def getDescenso(self):
      return ( self.s )
      

